"""
环境配置模块
"""
import os
from dotenv import load_dotenv

# 加载.env文件中的环境变量
load_dotenv()

# 环境配置
class Config:
    """基础配置类"""
    # 默认超时时间（秒）
    TIMEOUT = 30
    # 默认重试次数
    RETRY_TIMES = 3
    # 默认重试间隔（秒）
    RETRY_INTERVAL = 1

class DevConfig(Config):
    """开发环境配置"""
    # API基础URL
    BASE_URL = os.getenv('DEV_BASE_URL', 'https://dev-api.example.com')
    # API密钥
    API_KEY = os.getenv('DEV_API_KEY', '')

class TestConfig(Config):
    """测试环境配置"""
    # API基础URL
    BASE_URL = os.getenv('TEST_BASE_URL', 'https://test-api.example.com')
    # API密钥
    API_KEY = os.getenv('TEST_API_KEY', '')

class ProdConfig(Config):
    """生产环境配置"""
    # API基础URL
    BASE_URL = os.getenv('PROD_BASE_URL', 'https://api.example.com')
    # API密钥
    API_KEY = os.getenv('PROD_API_KEY', '')

# 根据环境变量选择配置
def get_config():
    """获取当前环境的配置"""
    env = os.getenv('ENV', 'dev').lower()
    if env == 'prod':
        return ProdConfig
    elif env == 'test':
        return TestConfig
    else:
        return DevConfig

# 当前环境配置
current_config = get_config() 